package com.aaa.boot.utils;

import com.aaa.boot.pojo.dto.PermDTO;

import java.util.ArrayList;
import java.util.List;

/**
 * @author MK
 * @version 1.0
 * 创建时间：2025/3/21 9:55
 */
public class PermissionUtil {
    public static ArrayList<PermDTO> handlerData(List<PermDTO> permDTOS){
        ArrayList<PermDTO> data = new ArrayList<>();
        for(PermDTO permDTO : permDTOS){
            if (permDTO.getPid()==0){
                data.add(permDTO);
            }
        }
        for (PermDTO permDTO : permDTOS) {
            Integer pid = permDTO.getPid();
            for (PermDTO datum: data){
                if (datum.getId().equals(pid)){
                    datum.getSubMenu().add(permDTO);
                }
            }
        }
        return data;
    }
}
